<script lang="ts" setup>
import * as echarts from "echarts";

import "echarts-gl";
import { onMounted, ref } from "vue";
let echartRef = echarts;
const chartEl = ref();
let chart: any = {};
let isSelected: any;
let isHovered: any;
let startRatio: any;
let endRatio: any;

let selectedIndex: any = "";
let hoveredIndex: any = "";

onMounted(() => {
  //   initEcharts();
  const colorList = ["#5A9CF1", "#6ED3D3"];
  const dataSource = [
    {
      name: "可用仓库",
      value: 75,
    },

    {
      name: "已用仓库",
      value: 25,
    },
  ];
  const paramsList = dataSource.map((item, index) => {
    return {
      ...item,
      shading: "realistic",
      itemStyle: {
        color: colorList[index],
      },
    };
  });
  // 传入数据生成 option
  let option = getPie3D(paramsList);

  // 是否需要label指引线，如果要就添加一个透明的2d饼状图并调整角度使得labelLine和3d的饼状图对齐，并再次setOption
  option.series.push({
    name: "", //自己根据场景修改
    type: "pie",
    hoverAnimation: true, // 悬停不放大
    label: {
      position: "bottom",
      formatter: function (params) {
        return `{percentSty|${params.percent}%}\n{nameSty|${params.name}}`;
      },
      rich: {
        nameSty: {
          fontSize: 16,
          lineHeight: 22,
          fintWeight: 400,
        },
        percentSty: {
          fontSize: 14,
          lineHeight: 20,
          fintWeight: 400,
          color: "#FFFFFF",
        },
        countSty: {
          fontSize: 14,
          lineHeight: 20,
          fintWeight: 400,
          color: "#B9D3ED",
          padding: [0, 8, 0, 8],
          backgroundColor: "rgb(90,156,241,0.3)",
          borderRadius: 2,
        },
      },
    },
    labelLine: {
      showAbove: true,
      length: 20, // 视觉引导线第一段的长度
      length2: 40, // 视觉引导项第二段的长度
      lineStyle: {
        color: "#686868", // 改变标示线的颜色
        width: 2,
        type: "solid", // 线的类型
      },
    },
    startAngle: 60, // 起始角度，支持范围[0, 360]。
    clockwise: true, // 饼图的扇区是否是顺时针排布。上述这两项配置主要是为了对齐3d的样式
    radius: ["40%", "52%"],
    center: ["50%", "53%"],
    data: dataSource.map((item, index) => {
      return {
        ...item,
        shading: "realistic",
        itemStyle: {
          color: "transparent",
        },
      };
    }),
    itemStyle: {
      opacity: 1, //这里必须是0，不然2d的图会覆盖在表面
    },
  });
  let chart = echartRef.init(chartEl.value);
  chart.setOption(option);
  chart.on("click", function (params: any) {
    // 从 option.series 中读取重新渲染扇形所需的参数，将是否选中取反。
    if (!option.series[params.seriesIndex].pieStatus) {
      return;
    }
    isSelected = !option.series[params.seriesIndex].pieStatus.selected;
    isHovered = option.series[params.seriesIndex].pieStatus.hovered;
    startRatio = option.series[params.seriesIndex].pieData.startRatio;
    endRatio = option.series[params.seriesIndex].pieData.endRatio;

    // 如果之前选中过其他扇形，将其取消选中（对 option 更新）
    if (selectedIndex !== "" && selectedIndex !== params.seriesIndex) {
      option.series[selectedIndex].parametricEquation = getParametricEquation(
        option.series[selectedIndex].pieData.startRatio,
        option.series[selectedIndex].pieData.endRatio,
        false,
        false
      );
      option.series[selectedIndex].pieStatus.selected = false;
    }

    // 对当前点击的扇形，执行选中/取消选中操作（对 option 更新）
    option.series[params.seriesIndex].parametricEquation =
      getParametricEquation(startRatio, endRatio, isSelected, isHovered);
    option.series[params.seriesIndex].pieStatus.selected = isSelected;

    // 如果本次是选中操作，记录上次选中的扇形对应的系列号 seriesIndex
    isSelected ? (selectedIndex = params.seriesIndex) : null;

    console.log("option-click: ", option);
    // 使用更新后的 option，渲染图表
    chart.setOption(option);
  });

  // 监听 mouseover，近似实现高亮（放大）效果
  chart.on("mouseover", function (params: any) {
    // 准备重新渲染扇形所需的参数
    let isSelected;
    let isHovered;
    let startRatio;
    let endRatio;

    // 如果触发 mouseover 的扇形当前已高亮，则不做操作
    if (hoveredIndex === params.seriesIndex) {
      return;
      // 否则进行高亮及必要的取消高亮操作
    } else {
      // 如果当前有高亮的扇形，取消其高亮状态（对 option 更新）
      if (hoveredIndex !== "") {
        // 从 option.series 中读取重新渲染扇形所需的参数，将是否高亮设置为 false。
        isSelected = option.series[hoveredIndex].pieStatus.selected;
        isHovered = false;
        startRatio = option.series[hoveredIndex].pieData.startRatio;
        endRatio = option.series[hoveredIndex].pieData.endRatio;

        // 对当前点击的扇形，执行取消高亮操作（对 option 更新）
        option.series[hoveredIndex].parametricEquation = getParametricEquation(
          startRatio,
          endRatio,
          isSelected,
          isHovered
        );
        option.series[hoveredIndex].pieStatus.hovered = isHovered;

        // 将此前记录的上次选中的扇形对应的系列号 seriesIndex 清空
        hoveredIndex = "";
      }

      // 如果触发 mouseover 的扇形不是透明圆环，将其高亮（对 option 更新）
      if (params.seriesName !== "mouseoutSeries") {
        // 从 option.series 中读取重新渲染扇形所需的参数，将是否高亮设置为 true。
        if (!option.series[params.seriesIndex].pieStatus) {
          return;
        }
        isSelected = option.series[params.seriesIndex].pieStatus.selected;
        isHovered = true;
        startRatio = option.series[params.seriesIndex].pieData.startRatio;
        endRatio = option.series[params.seriesIndex].pieData.endRatio;

        // 对当前点击的扇形，执行高亮操作（对 option 更新）
        option.series[params.seriesIndex].parametricEquation =
          getParametricEquation(startRatio, endRatio, isSelected, isHovered);
        option.series[params.seriesIndex].pieStatus.hovered = isHovered;

        // 记录上次高亮的扇形对应的系列号 seriesIndex
        hoveredIndex = params.seriesIndex;
      }
      // 使用更新后的 option，渲染图表
      chart.setOption(option);
    }
  });

  // 修正取消高亮失败的 bug
  chart.on("globalout", function () {
    console.log(hoveredIndex);
    if (hoveredIndex !== "") {
      // 从 option.series 中读取重新渲染扇形所需的参数，将是否高亮设置为 true。
      const isSelected = option.series[hoveredIndex].pieStatus.selected;
      const isHovered = false;
      const startRatio = option.series[hoveredIndex].pieData.startRatio;
      const endRatio = option.series[hoveredIndex].pieData.endRatio;

      // 对当前点击的扇形，执行取消高亮操作（对 option 更新）
      option.series[hoveredIndex].parametricEquation = getParametricEquation(
        startRatio,
        endRatio,
        isSelected,
        isHovered
      );
      option.series[hoveredIndex].pieStatus.hovered = isHovered;

      // 将此前记录的上次选中的扇形对应的系列号 seriesIndex 清空
      hoveredIndex = "";
    }

    // 使用更新后的 option，渲染图表
    chart.setOption(option);
  });
});

function getParametricEquation(
  startRatio: number,
  endRatio: number,
  isSelected: boolean,
  isHovered: boolean
) {
  // 计算
  let midRatio = (startRatio + endRatio) / 2;

  let startRadian = startRatio * Math.PI * 2;
  let endRadian = endRatio * Math.PI * 2;
  let midRadian = midRatio * Math.PI * 2;

  // 如果只有一个扇形，则不实现选中效果。
  if (startRatio === 0 && endRatio === 1) {
    isSelected = false;
  }

  // 计算选中效果分别在 x 轴、y 轴方向上的位移（位移均为 0）
  let offsetX = 0;
  let offsetY = 0;
  // 计算选中效果在 z 轴方向上的位移（未选中，位移均为 0）
  let offsetZ = isSelected ? 0.15 : 0;
  // 计算高亮效果的放大比例（未高亮，则比例为 1）
  let hoverRate = isHovered ? 1.05 : 1;

  let tmp = 0;

  // 返回曲面参数方程
  return {
    u: {
      min: 0,
      max: Math.PI * 2,
      step: Math.PI / 100,
    },

    v: {
      min: 0,
      max: Math.PI,
      step: Math.PI / 50,
    },

    x: function (u: number, v: number) {
      if (midRatio - 0.5 < 0) {
        if (u < startRadian || u > midRadian + Math.PI) {
          tmp =
            u - Math.PI - midRadian < 0
              ? u + Math.PI - midRadian
              : u - Math.PI - midRadian;
          return (
            offsetX +
            ((Math.sin(startRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
        if (u > endRadian && u < midRadian + Math.PI) {
          tmp = midRadian + Math.PI - u;
          return (
            offsetX +
            ((Math.sin(endRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
      } else {
        if (u < startRadian && u > midRadian - Math.PI) {
          tmp = u + Math.PI - midRadian;
          return (
            offsetX +
            ((Math.sin(startRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
        if (u > endRadian || u < midRadian - Math.PI) {
          tmp =
            midRadian - Math.PI - u < 0
              ? midRadian + Math.PI - u
              : midRadian - Math.PI - u;
          return (
            offsetX +
            ((Math.sin(endRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
      }
      return offsetX + Math.sin(v) * Math.sin(u) * hoverRate;
    },

    y: function (u: number, v: number) {
      if (midRatio - 0.5 < 0) {
        if (u < startRadian || u > midRadian + Math.PI) {
          tmp =
            u - Math.PI - midRadian < 0
              ? u + Math.PI - midRadian
              : u - Math.PI - midRadian;
          return (
            offsetY +
            ((Math.cos(startRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
        if (u > endRadian && u < midRadian + Math.PI) {
          tmp = midRadian + Math.PI - u;
          return (
            offsetY +
            ((Math.cos(endRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
      } else {
        if (u < startRadian && u > midRadian - Math.PI) {
          tmp = u + Math.PI - midRadian;
          return (
            offsetY +
            ((Math.cos(startRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
        if (u > endRadian || u < midRadian - Math.PI) {
          tmp =
            midRadian - Math.PI - u < 0
              ? midRadian + Math.PI - u
              : midRadian - Math.PI - u;
          return (
            offsetY +
            ((Math.cos(endRadian) * tmp) /
              (Math.PI - midRadian + startRadian)) *
              hoverRate
          );
        }
      }
      return offsetY + Math.sin(v) * Math.cos(u) * hoverRate;
    },

    z: function (u: any, v: number) {
      return offsetZ + (Math.cos(v) > 0 ? 0.1 : -0.1);
    },
  };
}
function getPie3D(pieData: string | any[]) {
  let series: Array<any> = [];
  let sumValue = 0;
  let startValue = 0;
  let endValue = 0;
  let legendData: Array<any> = [];
  // 为每一个饼图数据，生成一个 series-surface 配置
  for (let i = 0; i < pieData.length; i++) {
    sumValue += pieData[i].value;
    let seriesItem: any = {
      name:
        typeof pieData[i].name === "undefined" ? `series${i}` : pieData[i].name,
      type: "surface",
      parametric: true,
      wireframe: {
        show: false,
      },
      pieData: pieData[i],
      pieStatus: {
        selected: false,
        hovered: false,
      },
    };

    if (typeof pieData[i].itemStyle != "undefined") {
      let itemStyle: any = {};
      if (typeof pieData[i].itemStyle.color != "undefined") {
        itemStyle.color = pieData[i].itemStyle.color;
      }
      if (typeof pieData[i].itemStyle.opacity != "undefined") {
        itemStyle.opacity = pieData[i].itemStyle.opacity;
      }
      seriesItem.itemStyle = itemStyle;
    }
    series.push(seriesItem);
  }
  // 使用上一次遍历时，计算出的数据和 sumValue，调用 getParametricEquation 函数，
  // 向每个 series-surface 传入不同的参数方程 series-surface.parametricEquation，也就是实现每一个扇形。
  for (let i = 0; i < series.length; i++) {
    endValue = startValue + series[i].pieData.value;
    series[i].pieData.startRatio = startValue / sumValue;
    series[i].pieData.endRatio = endValue / sumValue;
    series[i].parametricEquation = getParametricEquation(
      series[i].pieData.startRatio,
      series[i].pieData.endRatio,
      false,
      false
    );
    startValue = endValue;
    legendData.push(series[i].name);
  }
  // 补充一个透明的圆环，用于支撑高亮功能的近似实现。
  series.push({
    name: "mouseoutSeries",
    type: "surface",
    parametric: true,
    wireframe: {
      show: false,
    },
    itemStyle: {
      opacity: 0,
    },
    parametricEquation: {
      u: {
        min: 0,
        max: Math.PI * 2,
        step: Math.PI / 20,
      },
      v: {
        min: 0,
        max: Math.PI,
        step: Math.PI / 20,
      },
      x: function (u: number, v: number) {
        return Math.sin(v) * Math.sin(u) + Math.sin(u);
      },
      y: function (u: number, v: number) {
        return Math.sin(v) * Math.cos(u) + Math.cos(u);
      },
      z: function (u: any, v: number) {
        return Math.cos(v) > 0 ? 0.1 : -0.1;
      },
    },
  });
  // 准备待返回的配置项，把准备好的 legendData、series 传入。
  let option = {
    legend: {
      show: false,
      data: legendData,
    },
    xAxis3D: {
      min: -1,
      max: 1,
    },
    yAxis3D: {
      min: -1,
      max: 1,
    },
    zAxis3D: {
      min: -1,
      max: 1,
    },
    grid3D: {
      show: false,
      boxHeight: 100, // 厚度
      top: 4,
      left: 0,
      boxWidth: 144,
      viewControl: {
        //3d效果可以放大、旋转等，请自己去查看官方配置
        alpha: 43, // 角度
        beta: 0, // 饼块开始位置角度
        rotateSensitivity: 0,
        zoomSensitivity: 0,
        panSensitivity: 0,
        autoRotate: false,
      },
      light: {
        main: {
          color: "rgb(255,255,255)", // 主光源的颜色。
          shadow: true, // 主光源是否投射阴影
          alpha: 80, // 主光源绕 x 轴，即上下旋转的角度
        },
      },
    },
    series: series,
  };
  return option;
}
// 监听鼠标事件，实现饼图选中效果（单选），近似实现高亮（放大）效果。

// 监听点击事件，实现选中效果（单选）
</script>
<template>
  <div class="chart-wrapper">
    <div ref="chartEl" :style="{ width: '300px', height: '300px' }"></div>
  </div>
</template>
<style lang="scss" scoped>
.chart-wrapper {
  display: flex;
  flex-direction: column;
  justify-content: center;
  align-items: center;
}
</style>
